package com.huike.clues.service.impl;

import java.io.IOException;
import java.io.InputStream;
import java.text.SimpleDateFormat;
import java.util.Date;
import java.util.UUID;

import javax.servlet.ServletOutputStream;
import javax.servlet.http.HttpServletResponse;

import com.alibaba.excel.util.StringUtils;
import io.minio.*;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import org.springframework.web.multipart.MultipartFile;

import com.huike.clues.service.ISysFileService;
import com.huike.common.config.MinioConfig;
import com.huike.common.core.domain.AjaxResult;

import lombok.extern.slf4j.Slf4j;

@Service
@Slf4j
public class SysFileServiceImpl implements ISysFileService {

    @Autowired
    MinioConfig minioConfig;

    /**
     * 文件上传至Minio
     * 使用try catch finally进行上传
     * finally里进行资源的回收
     */
    @Override
    public AjaxResult upload(MultipartFile file) {
        InputStream inputStream = null;

        MinioClient minioClient = getClient();//创建Minio的连接对象
        String bucketName = minioConfig.getBucketName(); //获取桶名

        try {
            inputStream = file.getInputStream();
            /**
             * 思路：我们上传的文件是:合同.pdf
             * 那么我们应该上传到配置的bucket内 我们配置的bucketName是huike-crm
             * 并且放在huike-crm/2021/05/05/这个目录下
             * 而对于同一个文件，存在重名问题，所以我们应该利用UUID生成一个新的文件名，并拼接上 .pdf 作为文件后缀
             * 那么完整的路径就是 huike-crm/2021/05/05/uuid.pdf
             */
            //TODO 基于上述逻辑补全代码
            //1. 文件名处理
            // 获取当前时间
            SimpleDateFormat sdf = new SimpleDateFormat("yyyy/MM/dd");
            String time = sdf.format(new Date());
            String filename = file.getOriginalFilename();
            if (!StringUtils.isEmpty(filename)) {
                filename = "/" + bucketName + time + UUID.randomUUID() + filename.substring(filename.lastIndexOf("."));
            }

            // 自动创建桶（如果不存在）
            if (!minioClient.bucketExists(BucketExistsArgs.builder()
                    .bucket(bucketName).build())) {
                minioClient.makeBucket(MakeBucketArgs.builder()
                        .bucket(bucketName).build());
            }

            //2. 文件上传
            minioClient.putObject(PutObjectArgs
                    .builder()
                    .bucket(bucketName)
                    .object(filename)
                    .stream(inputStream, file.getSize(), -1)
                    .contentType("pdf")
                    .build());

            //3. 构建返回结果集,将文件名(fileName)和访问路径(url)返回
            StringBuilder url = new StringBuilder(bucketName);
            url.append("http://").append(minioConfig.getEndpoint())
                    .append(":").append(minioConfig.getPort())
                    .append(filename);
            AjaxResult ajax = AjaxResult.success();
            ajax.put("fileName", filename);
            ajax.put("url", url);

            //4. 将结果返回
            return ajax;
        } catch (Exception e) {
            e.printStackTrace();
            return AjaxResult.error("上传失败");
        } finally {
            if (inputStream != null) {
                try {
                    inputStream.close(); // 关闭流
                } catch (IOException e) {
                    log.debug("inputStream close IOException:" + e.getMessage());
                }
            }
        }
    }


    /**
     * 免费提供一个获取Minio连接的方法
     * 获取Minio连接
     *
     * @return
     */
    private MinioClient getClient() {
        MinioClient minioClient =
                MinioClient.builder()
                        .endpoint("http://" + minioConfig.getEndpoint() + ":" + minioConfig.getPort())
                        .credentials(minioConfig.getAccessKey(), minioConfig.getSecretKey())
                        .build();
        return minioClient;
    }
}
